package algorithm.sort;


import java.util.Arrays;

/**
 * @author hck soccerlover9527@gmail.com
 */
public class KminArray {
    public static void main(String[] args) {
        int[] array = new int[]{4, 5, 1, 6, 2, 7, 3, 8};
        int k = 8;
        System.out.println(Arrays.toString(findKminArray(array, k)));
    }

    /**
     * 给定一个数组，找出其中最小的K个数
     *
     * @param array 原始数组
     * @param k     需要返回的k长度的数组
     */
    public static int[] findKminArray(int[] array, int k) {
        if (k > array.length || k <= 0) {
            return null;
        }
        //	数组下标从0开始
        k = k - 1;
        int len = array.length;
        int[] result = new int[k + 1];
        for (int i = 0; i < len; i++) {
            //	这里前面的已经有序不需要额外进行遍历对比
            for (int j = 0; j < len - i - 1; j++) {
                if (array[j] > array[j + 1]) {
                    swap(array, j, j + 1);
                }
            }
            //	第 i 趟排序完成，前面 i 个数都是有序的。所以对应需求中只需要找出最小的 K 个数，后续排序可以不用继续
            if (i == k) {
                System.arraycopy(array, 0, result, 0, k + 1);
                return result;
            }
        }
        return array;
    }

    private static void swap(int[] array, int i, int j) {
        int temp = array[i];
        array[i] = array[j];
        array[j] = temp;
    }
}
